/*******************************************************************************
 * This file is part of blueprint4j www.blueprint4j.org.
 * 
 *     blueprint4j is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Lesser General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 * 
 *     blueprint4j is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU Lesser General Public License for more details.
 * 
 *     You should have received a copy of the GNU Lesser General Public License
 *     along with blueprint4j. If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/
package blueprint4j.db;

import java.sql.*;
import java.util.*;

public class EntityEnumeration implements Enumeration {
	private Entity current;
	private Entity next;
	private int limit = -1;
	private int pos = 0;

	public EntityEnumeration(Entity p_entity){
		next = p_entity;
	}

	public EntityEnumeration(Entity p_entity, int p_limit){
		this(p_entity);
		limit = p_limit;
	}

	public boolean hasMoreElements() {
		return next != null;
	}

	public Object nextElement()
	throws NoSuchElementException {
		try {
			pos++;
			if (next == null) {
				throw new NoSuchElementException("No more Entities");
			}
			if (next == null || (limit > -1 && pos > limit)) {
				throw new NoSuchElementException("No more Entities was told to stop at " + limit);
			}
			current = next;
			next = next.getNextEntity();
		} catch (Exception de) {
			NoSuchElementException nsme=new NoSuchElementException(de.getMessage());
			nsme.setStackTrace(de.getStackTrace());
			throw nsme;
		}
		return current;
	}

	public Vector getVector()
	throws NoSuchElementException {
		Vector vector=new Vector();
		while (hasMoreElements()){
			vector.add(nextElement());
		}
		return vector;
	}
}

